Add commented out model switching code