Prateek's submission of BRAINWAVE | CodeChef