Adding my solution for assignment 1