Python - Shape Calculator