Refactored head unit and brake models into a shared model