My initial code