Problems with the graphics in .py