Fixed some code styling