Added doc, improve plot script