Add to tests section