Cleaning up the code just a little bit.