still working on the initial code