Hello! Could you please share the code to reproduce PCA plot from the paper? I am unable to reproduce this result. Thank you!