Imported from https://github.com/YuhangSong/Arena-Baselines