prob 4b py