Clean up the code for a little bit.