updated readme and code