Do you think the ending was a dream sequence?
No way the ending could be THAT happy and he get out of jail after only 5 years, a rainbow in the background and they just drive off....
It all had to be a dream Baby had while in jail.