update code

Signed-off-by: Thanh (Adam) T. NGUYEN <adamnguyen.itdn@gmail.com>