Added visualization for the other two classes