Add tests to testthat folder