Add error trap handling