Enhanced the function to extract data and plot it. Now the function returns a plot with various subplots to compare