First part of the problem is done. Testing included. Not sure if testing the output is appropriate as it contains print statements.