clean up a little code