Small changes to code indentation