Add error catching