Add BaseUnit class