add the code support the entry_point
add the support for entry_point