added a little documentation in the code