Rework visualization to render function bodies

Also, use distributed vis functionality even in the non-distributed
case, to cut down on code duplication.