added visualization for outputs