Add exception code

It would be nice, to add exception code too.
Would be nice to have a space here as well after the `:`, and adjusting the tests as well.