Add initial working test