updated figure to include equations