code for 3331 python programming