edited drive code