Implemented unit label feature with unit tests