prepared code to add categories and added comments for the future