Minor changes
fixed figre 4