Clean the code, do some re-struct thing.