Edit to square function
plot y = cos(square(x))
Added label and title
Changed to OOP way of defining plots