Working on remove task function