Cleaning up a little code.