Update : Remove unused class