Edited the sqare function.