Getting wrong answer on vivek and party