diff --git a/fps_example.py b/fps_example.py
index a22f0d6..d53ac36 100644
--- a/fps_example.py
+++ b/fps_example.py
@@ -12,26 +12,29 @@ def main():
 
     while True:
         ret, frame = cap.read()
-        frame = imutils.resize(frame, width=800)
-        total_frames = total_frames + 1
+        if(ret):
+            frame = imutils.resize(frame, width=800)
+            total_frames = total_frames + 1
 
-        fps_end_time = datetime.datetime.now()
-        time_diff = fps_end_time - fps_start_time
-        if time_diff.seconds == 0:
-            fps = 0.0
-        else:
-            fps = (total_frames / time_diff.seconds)
+            fps_end_time = datetime.datetime.now()
+            time_difference = fps_end_time - fps_start_time
 
-        fps_text = "FPS: {:.2f}".format(fps)
+            if time_difference.seconds ==0:
+                fps=0.0
 
-        cv2.putText(frame, fps_text, (5, 30), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (0, 0, 255), 1)
+            else:
+                fps = total_frames / time_difference.seconds
 
-        cv2.imshow("Application", frame)
-        key = cv2.waitKey(1)
-        if key == ord('q'):
-            break
+            fps_text = "FPS : {:.2f}".format(fps)
 
-    cv2.destroyAllWindows()
+            cv2.putText(frame , fps_text,(5,30),cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (0,0,255),1)
 
+            cv2.imshow("Application",frame)
+            key = cv2.waitKey(1)
+            if key == ord('q'):
+                break
+        else:
+            cv2.destroyAllWindows()
+            break
 
 main()