Improved arena detection for squares