add tests (broken atm)