The current if statement doesn't detect any particle that has a negative z-component of the momentum and are pushed into it.