added shit to main.py