Update generator
Bump version to 0.6.1