I installed the SyntaxFold package and customized the key binding. There needs to be 2 empty lines between sub/headings for folding to work.

It's not the prefect solution but good enough.