Code cleaning - rubocop