updaed code and readme