Add basic train animation