clean up initial code