Fix some bugs, set stage for peak partitioned TSNE