Category - help with python homework