some code cleanup

Signed-off-by: DarkSnake-Kobra <jbttoms@gmail.com>