Added return value to arguments list of after function if the return value is defined