Top with more polish. Imagine seeing the burning of the tower and the resurrection of the Beasts in person, Celebi maybe brought you back because of ____, have some more towns, cities, trainers and points of interest, keep abilities and held items, it would be the next logical step in the Legends games
I did just think, maybe all the battle stuff omitted from PLA will be re-added in it updated form in each game like how they originally developed, e.g Legends 2 has held items, Legends 3 has abilities, etc etc