Merge pull request #2 from radibnia77/main

Add tests