Modified gem to include Rails 3 support