增加category，并修复一些细节问题