Display updated, geometry needs fixing. Pygame window also does not close