basic program in python for interview