added test/train folders