Review original repo

I changed some things to make the code a better representation of idiomatic Ruby.