Final modificaiton of source code before submitting
Added the final submitted solution to repo