moved TransportProblem class outside main file and cleaned