Add main method.

Signed-off-by: RyeYao <rye.y.cn@gmail.com>