Code styling fixed