Remove test directory and just have tests