Add generic error catch