rm unused argument in visualization function