My first code in python