Ideally, you should play Episode Prompto of Final Fantasy XV once you reach Chapter 12 of the game. That allows you to follow the complicated events between Prompto falling off the train and his reunion with his friends in Niflheim.