Add expansion pass