hw5 progress on problem 2