Added figure, other small changes.