Added support for easy visualization testing Ref #20