Refactor/cleanup

Signed-off-by: Brandyn A. White <bwhite@dappervision.com>