Updated code per coding style