Modified register task logic