Adding viz, test, python