Remove old files, Bugfix for class weight failing