Changed get_unit to return references, as an optimization and so that training would show up properly for Actions