Adding documentation to code

- issue #14