Zhihao Bai, Johns Hopkins University