Added visualization for counts vs features