I have added basic pygame code to main.py