Iterated on feedback from AndreM.

- Removed "moment"
- Input and output explicit now by split into input_ and output_.
- Added flags