First coding exercise in phython