Changed UnitToken's edges storage method to map.