added testing code to main.py