Small refactoring and added documentation