Add the handle of the argument "removeAfter" and the test to verify if it work.