Hekang Yang, Columbia University