finished the initial code