Added a legenda and added colors