simplified main.py