Add pygame window and loop