Adds error handling

kpmartin