FoX: Formation-aware exploration in multi-agent reinforcement learning