Add unit class