Did you think that the train would be about using drugs?

Joy doesn't need to walk to HQ, she can use a train to get there faster = use drugs to quickly achieve happiness

Going through that part where they change into 2D shapes = psychedelic drugs

It would make perfect sense

A missed opportunity

Maybe in the sequel