Add doc strings to viz module