clean up code and add readme