Add the restore focus feature
Remote unused function