Python Octahedron Program