working on prob 4 proj 2