updating unittest
Merge pull request #16 from Henry-Benito/master

updating unittest