Updated to add legends