Add support for exit codes