Chaning Ui for classfier, removing unwanted things