fix bug in trace + better pngs