Remove document classes TODO