Add function to calculate normalized ws_diff plots