To update environment according to the new version of gym

Some deprecation warnings appeared for the latest version of gym.