Adding square. Getting an error on each in the second function. Need to resolve.