Fixed categories and added method

Fixed categories and added method