edited typos for nips paper