Categories feature now works