fix bug
fix bug
Merge pull request #6 from Nazgard/tests

Tests