Small refactor
Documentation