simplified main.py to reflect changes