Add error check