Minor adjustments
Quick fix