working visualization from the python program