Add my first python code