Thanks a lot for such a wonderful code! If I hope to run MELD in other image-based environments like DMControl and Atari, how should change the code? Thanks a lot!