code clean up and read me update