Square.py

Exercise 5: Updated