Switched to SquareEnvironment instead of toy Environment.