Added more functionality. I still can't run it myself due to cv2 problems