add square func