added option to visualize TSR