Adds comments for train files