Minor fix and README.md update