add first mock tests
add refactor of visualizations