Fixing styles based on rubocop