Add error output handling