Rewrite using a class representing the Drive unit