Add error hanlding