Updates to registration function

Signed-off-by: Zoe Rooney <zoe@zoerooney.com>