Initial python code