Added class idea for the Wedding problem