AIML331 Assignment 1