Add test suite

Adds a test suite to all existing functionality.