added python script to visualize results (only first iteration)