Bug fixes after testing with Florian

README.md

@@ -27,7 +27,7 @@ command in the command line. Since this repository is private, you need to add y
 
 ```
 git clone https://gitlab.ruhr-uni-bochum.de/mdi/xps-pred.git
-cd xps_prediction
+cd xps-pred
 ```
 ...and install it with pip:
 

src/xps_prediction/plotting.py

@@ -32,7 +32,7 @@ def plot_composition(comps: pd.DataFrame, coords: pd.DataFrame, train_ids: np.nd
         # Plot the data points
         ax[i].scatter(coords.iloc[:, 0], coords.iloc[:, 1], c='lightgrey', s=66)
         # Plot the data points
-        sc = ax[i].scatter(coords.loc[comps.index].iloc[:, 0], coords.loc[comps.index].iloc[:, 0], c=comps[elem],
+        sc = ax[i].scatter(coords.loc[comps.index].iloc[:, 0], coords.loc[comps.index].iloc[:, 1], c=comps[elem],
                            edgecolors=e_colors, s=66, linewidth=1.5, cmap=cmap)
         # Plot a colorbar
         divider = make_axes_locatable(ax[i])