code refine

Signed-off-by: shyeon.kim <kimseo@kth.se>