Cleaned up plot.py.
Added unit tests.
Added xlim and ylim functions.