AIML331_Assignment1