Test it

Add tests; They're OK... ok not brilliant.