updated code and README