First attempt at adding some testing