add solution to assignment 1