Merge pull request #15 from riachkyle/Nick

synched API with visualization
updated API with tests