Add support for four main category