Fix bug; add visualization interface