Address rubocop feedback