prob3 solved partially