Fixes the header
Adds switch