the 'square' example has been updated