added hw2 prob 5