Modified the code to add functions and arguments