Kind of working with timeout code