Added documentation to the added functionality