made a basic screen structure for handling the UI with pygame