Small changes.
Fixed plotting scirpt.