CATEGORY help with python homework