Updated the switch styles to match the new element