PSR code tidy