Add extract function