my initial code!