Updated to remove vocabulary generation code since we don't need to do our weird max calculation anymore.