Added the argument to the exit function