Simple graphics in Python