Learning to Apply Theory of Mind