Hello, thank you for your awesome work! The SE paper mentions cleaner code, where can one find the repo?