Scale back code to match lesson