tidied up code with rubocop
Merge pull request #2 from BenMorganIO/rubocop

tidied up code with rubocop