Start adding basic tests