Increase testing / coverage

Purpose
-----------
Improve code coverage of tests.

Changes
------------
* Added tests
* Update docs

Side Effects
----------------
Fixed potential bug with image exporting that testing exposed