Refactored code and deleted unnecessary tests in Train and Station classes