Modify prop function to remove noise

After training remove added Poisson noise