Add timeout option