Added a square method with some user input! Hope that's ok Kantar.

Testing out a pull request