adding more tests; moving plotting methods to separate module; configuring plotting methods to support three dimensions