I am currently working on a version with the improvement. I will upload it when the submissions are over.
