Add first version of tests