Add tests to functions

This will take some time and we need to define which tests.