Can just use a map