Trying out some basic programs in python