Cleaned up more code

Signed-off-by: Zane M. Kolnik <zanematthew@gmail.com>