Allow command function to take arguments