worked on test.py file
need to clean this up