Added annotation to matplotlib graph testing