Learning to reason with neural module networks