First attempt at tests