I was working through the lectures for the Foundations of Machine Learning tutorial and cloned the Jupyter notebook into my local instance of VS Code. Simple fix is to add using Statistics before mean function is called. Will submit pull request shortly.