update initial code