First version of test for each part as well as the whole function