updated square example