Cleaning things up for rubocop