refactor the code, clean the code.