Some code cleanup...

Signed-off-by: Chris Conley <chrisc93@vt.edu>