Cleans-up the code.