Govt to improve public transport