Small cleanup and README update