#4 Add error handlers