Add style and minor fixes

These two lines should be indented