add necessary classes