Oops! Forgot the new label classes