Slight code cleanup
Add RubyGems badge to readme