First steps towards unit tests