Style changes to pass Rubocop