Finishing up week 5 homework