added variables, chagnes in visualization