main.py for test is updated. Testing resulted included.