Made script actually work with real csv file