Fixed some rubocop issues