better format

add 2 little tests