Minor bug fix and README changes