Merge pull request #9 from saicheems/update

Update to reflect codegen