Made it work for defining and using the identity function