clean up before hw 3
clean up before hw 3