Started adding some tests

It is not much, but it is a start for solving #4 
Can you remove unneeded empty lines? Lines 9, 11 and 15.