Mini optimization & fixes