Update: Clean the code.