Code for analyzing caltrain data