clean up as i resume writing code
Exercise 3, train problem with many companies