Updated comments

Changed comment on legend

#8