add initial solution for homework 2