I'm not great with the rubies, lol.

I've only been working with ruby < a year, so I'm still learning to do things like this. Updated the PR.