add solution to assignment 2