Need to add some phpdoc for the new argument