PSR2 tidy up