Adding correct label classes