My solution to assignment 2