Adding functionality for the registration