how to create python program using function