prob 4a py