ps2b.py contains the solution to problem 3