Added data gen code.