Updated source code

Updated source code is submitted on the Github