lecture.2 initial code