quite hacky still but very simple classes are working