## Issue description ## Tasks - [ ] Train PNet with including both HCal hits - [ ] Include track states and extrapolated track states at ECAL - [ ] Compare to vanila PNet ## Existing material to start off * Source 1 * Source 1