Added partition logic
adding custom paritioner code to the project