add small tests