add in some categories logic