Minor changes to drawGraph