added visualization script