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