Can we use just className instead of dataset?