add more comments
fix header name
Merge pull request #31 from adrianob/unit_tests

Unit tests