added test.py for testing