Modified how to return from subsection to categories