Added a mechanism to add color to figure