add code and updated readme