Update features notebook

1. Add an example of `Context.register`
2. Add an example of running with MPI-only
3. Add and edit some comments and explanatory text