Graph UI improvements