Fixed some rubocop concerns.