Public transit optimization