Add unit class and complete CRUD