Add error testing