Updated for ruby 2.1.0