updated 'square' demo