Documentation, tests

Added documentation and some tests/ plots to the script