Think before you act: A simple baseline for compositional generalization