Added a solution for assignment 2