Add register function