Minor changes to code styling