Cleaned comments
Fixed viz code