removed unused class, minor fix