small refactor
Merge pull request #9 from Ravenstine/keygen

small refactor