Small tweaks

Fixed indentation issues