just some code clean up!

was getting dirty!