YH-137: Code cleanup