Fix docs a bit