Yes, I did not know how to transition back to the main screen and the code was a mess so I kind of left it as is :P
EDIT: Ending is now fixed and lag is reduced.