Started adding some tests

It is not much, but it is a start for solving #4