Finishing up homework for week 5