Adding solutions to assignment 1