How to use functions in python