Added plots.
Moved plotting features in separate module 'plot.py'.