Small refactoring and docs added