Adds checks for train and track