I added some tests