Polish up Lego code and slides