Minor fix and update on readme