basic programs in python for interview