update code
Signed-off-by: Jan Kristanto <jan@ontelstudio.com>