added hw2 prob2