Refactor code for the two headsets