add On_Load function

Signed-off-by: Wenzhao Fang <wzhfang@marvell.com>