Added the labels back and fixed a typo in the documentation