Started adding some tests

It is not much, but it is a start for solving #4 
Please leave a space before and after `=`. Same applies for lines 26 and 34.