Tests

Add tests to the project