Added comments for classify & distribution