Add validation error handling