have 5 classes,fix a bug