Add train button handler.