Add a basic test to check its all working