I definitely would have done the Awakening thing differently. I was actually quite liking it at first, when Optimus's spirit was standing there in front of "The One/Primus/Whatever". It was a very interesting mirror of Megatron's interaction with Unicron from the movie. I like the general idea of Transformer God convincing Optimus that it was time to stop fighting and move on, but not as in "you need to abandon the war" but instead "your time is over. I would have had Optimus come to realize this through some sort of less confusing vision quest, then ended the short at the moment that Rodimus picks up the Matrix inside of Unicron. Basically here at the end, we would see Optimus now watching events play out in the "real" world, and from the spirit world, he speaks out to Rodimus with the "Arise, Rodimus Prime" line, then lets his spirit pass on to the Matrix/Cybertron/Primus/what-have-you.