Add/fix code docs