OCR module refactor