Add unit and army classes
fix const ref