Improve the bug visualization