Rewriting the model for the drive system