Improving Multi-Agent Cooperation using Theory of Mind