rewrite of wheel drive function