[Update]: Update code structure.