fixed some small visualisation problems