Set up unit tests to ensure all functions work
Set up unit tests to ensure all functions work