Add countdown back in