Added comment and fixed lint