Minor tweak to code and slides