pygame 2 screen display not working properly : r/pygame