Of course I can share my code with you. I can add some more description as well if you’d like. What environment are you using? I wrote the code as a regular py file in spyder, but I could make a notebook for jupyter out of it.
Let me know what works best for you.