[WIP] [#11] Correct exit status code

Fixes #11

Need to do some testing before merging this