Argument

Changed the argument to test the answer