Add error control