Add function for plotting wake profile