Added docs, removed matplotlib import