добавив функцію для квадрату

добавив функцію для квадратучячччч