Styling fixes
Rubocop