add unit class