my initial code