Added comment and formatted code