Make exit code optional
Fixed bug with plain
Added tests for error handler
Added some more tests
Improve ui