updated drive code