Indentation fix and better test main