Testing square function