Updated code style

Better code styling