Added more features to functions

You can now expand the body of a function