Python Code For Shapes