Add timeout limit