Refactored the documentation part