Changed some code styling