removed unneeded bits from main.py