Add first working test