Add basic unit tests
Merge pull request #2 from StevenDevooght/add_unit_tests

Add basic unit tests