Add error checks