Improve drive output