Drive-train planning