added missing figure and cleaned up a bit