A small python program