Move plot routines from test.py to plot.py
Added docstrings to test.py and utils.py. Split features section into subsections