python program for if loop