fixed section # in homework 6 squares.py