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