Added test setup. 36% coverage now :-)