Solution for Assignment 1