add visualization routine