Adding Ankit's code to the repo