updated readme and the code