Add header
Fix rubocop issues