Add some documentation to the code