Add error handleing