Update initial code