Improving legend

app/src/main/java/net/pokeranalytics/android/ui/fragment/GraphFragment.kt

@@ -92,13 +92,15 @@ class GraphFragment : PokerAnalyticsFragment(), OnChartValueSelectedListener {
 
 		dataSetList.firstOrNull()?.let { dataSet ->
 
-			this.legendView.prepareWithStat(this.stat, dataSet.entryCount)
-
 			// initialize chart
 			this.chartContainer.removeAllViews()
 			this.chartView = when (dataSet) {
-				is LineDataSet -> LineChart(context)
+				is LineDataSet -> {
+					this.legendView.prepareWithStat(this.stat, dataSet.entryCount)
+					LineChart(context)
+				}
 				is BarDataSet -> {
+					this.legendView.prepareWithStat(this.stat)
 					val barChart = BarChart(context)
 					if (stat.showXAxisZero) {
 						barChart.xAxis.axisMinimum = 0.0f

app/src/main/java/net/pokeranalytics/android/ui/view/LegendView.kt

@@ -49,7 +49,7 @@ class LegendView : FrameLayout {
 	/**
 	 * Set the stat data to the view
 	 */
-	fun prepareWithStat(stat: Stat, counter: Int) {
+	fun prepareWithStat(stat: Stat, counter: Int? = null) {
 
 		if (stat.significantIndividualValue) {
 			this.stat1Name.text = stat.localizedTitle(context)
@@ -60,9 +60,12 @@ class LegendView : FrameLayout {
 			this.stat2Value.isVisible = false
 		}
 
-		val counterText = "$counter ${context.getString(R.string.sessions)}"
+		counter?.let {
+			val counterText = "$it ${context.getString(R.string.sessions)}"
 			this.counter.text = counterText
 			this.counter.isVisible = stat.shouldShowNumberOfSessions
+		}
+
 
 	}