Set up basic main.py and created my display.py