State.After requires state to be returned

[Moizsohail]

The following code throws an exception

const After = (state) => { console.log(state) };

The following code fixes it

const After = (state) => {
 console.log(state)
 return state;
};

please update documentation or update the code.

BTW: Love this project. Keep it up.