PTI returns to National Assembly