Imported from https://github.com/abhigoyal1997/CS-763-Project