My solution

My solution for assignment 2