Adde error handling