Add documentation

Add documentation describing the code