A pygame wrapper to simplify the main loops