Add user management code