Moving test to tests dir