Small update on this, I fixed a bug on the final stage with the music not playing/game freezing.

Please let me know if there are any further issues. Thanks!