Added PI Stage integration