added some visualization tricks