renamed test.py main.py and created new test file
implemented successful test
add more tests
fixed typos
created main function