Remove document classes, use repository in controller