Fixed function composition bug.  Now can use the square function to actually square things.