updated lecture04 code.