Repair tests.py
fixed it so I could test it
Add tests for the program