updated the code and the read me