fix square functionality.