Add gen code