working on colors and legend