Creating the Party Problem mode (based on chapter 6 of tutoriall