prob 1 modofied