fix a bug
Merge branch 'master' of https://github.com/QingxianLai/DSGA1007_Final_Project