Add millisecond timeout support