Add error checkign