My solution for assignment 2.