- Added main() function to test.py