Improve docs a bit