Fixes missing semicolon and code clean-up.

Signed-off-by: hegdeashwin <ashwin.hegde3@gmail.com>