adding comments
refactoring