First attempt at writing tests