Fix pause function

Now returns the argument it was given, rather than None.