Code style tidy up