Refactored Train tests; included runner file