Add error message handling.