Add comments / refactor code for readability.