Added classes for drawing node shapes