I had up to stage 3 done at some point but I can't seem to find that file. I'll just leave this up in case anyone wants to try and do an improvement for this in the future.