Fix code docs a bit.