Category functionality added