added condition to display category