This code is forked from this repo. Some functions from here are added. The code has been modified by myself for CSE559 at Washington University in St. Louis, Fall 2020.