My first pull request :)

Small changes to generators