add a part in a more understandable way.
code tidy-up