https://ketajaman.com/cara-menggunakan-python-square-function