Put this functionality into Registration model