Add timeout method