Pygame fiddly bits. Draw a square