Python Cube Code