add in some tests i forgot