Improve drawing of trace