Added drive functionality