Improve documentation, comment and README