Added new transit logic