Started to add visualziation