prob 2 py