Fixed an issue with a readme sample.
Updated handling of after to allow override of return value.