add visual goodies, fix bug