Focus pygame window

Otherwise it will not receive events.