Add Files

Added the python code so far