First Commit- Added the python code