adding some tests :)