Add error checking