Create test

Add unit tests for the main parts of the project to ensure integrity and durability.