Added registration functionality