add main function

add main function