Small cleanup and better readme