Tests adding

Simple tests added.
First PR on github try.