code visualization: multiple files

Signed-off-by: Fei Guan <feiguan@172-16-185-237.dynapool.nyu.edu>