Cleanup Code

Signed-off-by: riptidewave93 <chrisrblake93@gmail.com>