Yeah ending it on the climatic dance scene was the perfect movie ending, I don't really see the point of flash-forwarding six months or whatever to confirm that Johnny and Baby did make it work, imo they only added that so that they could claim that it does offer something more than the original