[FIX] add legend