add axis labels

basics.py

@@ -28,6 +28,7 @@ def plot(data, countries):
 
     for ax, fig, name in [(tc, tcp, "total_cases"), (td, tdp, "total_deaths"), (nc, ncp, "new_cases"), (nd, ndp, "new_deaths")]:
         ax.set_yscale("log")
+        ax.set_ylabel(name)
         for tick in ax.get_xticklabels():
             tick.set_rotation(45)
         ax.legend(frameon=False)