Learning how to write python