main.py

Main code for the test