Error handling enhaced