Add registration functionality