Code cleanup

Signed-off-by: Daniel Pierce <paladin2350@gmail.com>