updated 'square' example