added code and updated readme