Add error dectection