Add timeout error catch