Add working test