Rubocop'd

Updated a couple of generator files with rubocop compliant changes.