oops tried to be fancy, here's the fix to prob 4