Cleanup of visualization