Add code for the 3rd lecture