Style of code updated