small refactoring
Merge pull request #75 from learning-layers/smallRefactoring

small refactoring