Update sonogram.py

sonogram.py

@@ -255,6 +255,8 @@ class Sonogram():
                     if tracking:
                         singleDimList.append((currMembers,Segment(start,start+duration)))
                         currList.append((currMembers,Segment(start,start+duration)))
+                        if start+duration > endTime:
+                            endTime = start+duration
                         start = None
                         currMembers == None
                         duration = 0
@@ -262,6 +264,8 @@ class Sonogram():
             if tracking:
                 singleDimList.append((currMembers,Segment(start,start+duration)))
                 currList.append((currMembers,Segment(start,start+duration)))
+                if start+duration > endTime:
+                    endTime = start+duration
             categorySegmentList.append(currList)
         # Check where we left off
         if endTime != maxTime: