Added Drive Fix

Fixed Drive code. ALMOST.