Add in a basic test