Add callback to auth module