got class working