Modified Train and Station classes