Improve docs and readme