py file prob 2