Improve documenatation