InternetSpeedTest from Balena Examples - no longer functioning?

Reference this project: balena-io-examples/internetspeedtest,

It’s clearly an older project but a great example to start out with and deploy across site to check for speed issues etc. Unfortunately the speedtest doesnt actually run, citing the below error:

ImportError: cannot import name 'Literal' from 'typing' (/usr/local/lib/python3.7/typing.py)
 speedtest  
 speedtest  During handling of the above exception, another exception occurred:
 speedtest  
 speedtest  Traceback (most recent call last):
 speedtest    File "./speedtest.py", line 5, in <module>
 speedtest      import paho.mqtt.client as mqtt
 speedtest    File "/usr/local/lib/python3.7/site-packages/paho/mqtt/client.py", line 51, in <module>
 speedtest      from typing_extensions import Literal  # type: ignore
 speedtest  ModuleNotFoundError: No module named 'typing_extensions'

Just starting out with Balena Cloud etc and so no doubt I will use this project as a basis to building my own version but if anyone has any quick tips on getting this up and running its appreciated. Thanks