minor changes and indentation fix