Add small tests
READMEEE