Added necessary code in the files