Minor adjustments to distribution