Modified source code to pass the PSR2 check