Final PSR2 fixes