Optimization/Refactoring

Edited the way probability is created, is now much faster.