writing programs in python