Change code style to suit @phuu :)