Minor changes to figure for ex2