Fixed Rubocop issues