add categories fix #5