Updated generator. Fixes #2