prob 4 first attempt