more bug fixes
Merge branch 'master' of github.com:jieunchoi/MIST_codes