Add small testing framework