Fix code docs