some small changes in visualization