First non trivial python program