Cube Function In Python