modified draw() to color by element and also set radius according to element