Update main.py

minor changes for testing purposes