Code cleanup.

Signed-off-by: Toha <tohenk@yahoo.com>