Few code cleanups

Signed-off-by: zeroblade1984 <zeroblade1984@gmail.com>