Add header and fix rubocop issues