i just added some tests