Updated to Switch Statement

Changed the party class function to use a switch statement