main.py - only file needed