Simple graphics engine - Python