Updated code for Ruby 2.1.0