Fix code to match desc. + Gemfile