Modify import class