[enh] start to work on class (NOT WORKING YET)