Learning Pygame with a mathematical game