Refactor distribution algo to UE class