Python - Surface Area of 3D Shapes