updated readme, code clean up