added the missing visualization function