Add the correct code that doesn't time out