Update 02.4.md

Small changes to code.