Fixing Pervious Error

This code is to replace the original files that I mistakenly uploaded
before which didn't include some of the files as well as had old version
of the code