correction on 3 and 4

forgot to add the legend