Added comment and fixed code layout