Rename partition class and add scala doc