cleaning the code.